package ustc.xiaoyang.karl.doublepointer;

import java.util.*;

/**
 * 三数之和 15 medium
 */
public class TheSumOfThreeNumbers15 {
    public static void main(String[] args) {
        Set<List<Integer>> set = new HashSet<>();

        List<Integer> list1 = new ArrayList<>();
        list1.add(1);
        list1.add(2);

        List<Integer> list2 = new ArrayList<>();
        list2.add(1);
        list2.add(2); // 和 list1 顺序、内容完全相同

        List<Integer> list3 = new ArrayList<>();
        list3.add(2);
        list3.add(1); // 和 list1 内容相同但顺序不同
        list3.sort(Comparator.reverseOrder());
        set.add(list1);
        set.add(list2);
        set.add(list3);

        System.out.println(set.size()); // 输出 2（list1 和 list2 被去重，list3 视为不同）
        List<List<Integer>> res = set.stream().toList();
    }

    public List<List<Integer>> threeSum(int[] nums) {
        Set<List<Integer>> set = new HashSet<>();
        Arrays.sort(nums);
        for (int i = 0; i < nums.length; i++) {
            int left = i + 1;
            int right = nums.length - 1;
            while (left < right) {
                int sum = nums[i] + nums[left] + nums[right];
                if (sum == 0) {
                    set.add(Arrays.asList(nums[i], nums[left], nums[right]));
                    left++;
                    right--;
                    continue;
                }
                if (sum < 0) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return set.stream().toList();
    }
}
